Beyond Expert Rankings: Contrast Calibration for Mixed-Precision MoE Quantization
超越专家排序:混合精度 MoE 量化的对比校准
🧭 MoE 混合精度量化与敏感度校准 ★ 深度解读
Nanjing University of Posts and Telecommunications · Tylogi AI Lab / TAIL · Southeast University · Microsoft · University of Cambridge
NatScale 用对称 KL 目标校准专家敏感度对比度,在固定存储预算下提升 MoE 量化保真度
📌 工作参考价值(4 点)
- 部署 MoE 大模型(如生成式推荐、精排 reranker、Agent 策略模型)做混合精度量化时,不要只看专家重要性排序:用 power transformation 校准敏感度对比度,可将不同层/模型放在共同信息尺度上,默认 τ=1 nat 是无调参的实用起点。
- 工程实现上把 joint gate/up/down 误差、结构化先验和校准权重解耦并预缓存,之后扫预算或 τ 只需重跑线性规划分配器,无需额外前向/反向,适合线上反复调整内存配额。
- 分配单元下探到 expert 内部 projection,而不是整层统一精度,可在同层内差异化 bit-width,压缩比与效果更优;电商推荐模型若含 MoE 或专家模块可沿用同一思路。
- 注意 PPL 与 native KLD 可能偏好不同 τ:若目标是输出分布对齐(如广告创意生成、query 改写),优先看 KLD/Top1 而不是 PPL。